package com.sheng.leetcode.year2023.month02.day09;

import org.junit.Test;

import java.util.HashMap;
import java.util.Map;

/**
 * @author liusheng
 * @date 2023/02/09
 * <p>
 * 1797. 设计一个验证系统<p>
 * <p>
 * 你需要设计一个包含验证码的验证系统。每一次验证中，用户会收到一个新的验证码，<p>
 * 这个验证码在 currentTime 时刻之后 timeToLive 秒过期。如果验证码被更新了，<p>
 * 那么它会在 currentTime （可能与之前的 currentTime 不同）时刻延长 timeToLive 秒。<p>
 * 请你实现 AuthenticationManager 类：<p>
 * AuthenticationManager(int timeToLive) 构造 AuthenticationManager 并设置 timeToLive 参数。<p>
 * generate(string tokenId, int currentTime) 给定 tokenId ，在当前时间 currentTime 生成一个新的验证码。<p>
 * renew(string tokenId, int currentTime) 将给定 tokenId 且 未过期 的验证码在 currentTime 时刻更新。<p>
 * 如果给定 tokenId 对应的验证码不存在或已过期，请你忽略该操作，不会有任何更新操作发生。<p>
 * countUnexpiredTokens(int currentTime) 请返回在给定 currentTime 时刻，未过期 的验证码数目。<p>
 * 如果一个验证码在时刻 t 过期，且另一个操作恰好在时刻 t 发生（renew 或者 countUnexpiredTokens 操作），过期事件 优先于 其他操作。<p>
 * <p>
 * 示例 1：<p>
 * 输入：<p>
 * ["AuthenticationManager", "renew", "generate", "countUnexpiredTokens", "generate", "renew", "renew", "countUnexpiredTokens"]<p>
 * [[5], ["aaa", 1], ["aaa", 2], [6], ["bbb", 7], ["aaa", 8], ["bbb", 10], [15]]<p>
 * 输出：<p>
 * [null, null, null, 1, null, null, null, 0]<p>
 * <p>
 * 解释：<p>
 * AuthenticationManager authenticationManager = new AuthenticationManager(5); // 构造 AuthenticationManager ，设置 timeToLive = 5 秒。<p>
 * authenticationManager.renew("aaa", 1); // 时刻 1 时，没有验证码的 tokenId 为 "aaa" ，没有验证码被更新。<p>
 * authenticationManager.generate("aaa", 2); // 时刻 2 时，生成一个 tokenId 为 "aaa" 的新验证码。<p>
 * authenticationManager.countUnexpiredTokens(6); // 时刻 6 时，只有 tokenId 为 "aaa" 的验证码未过期，所以返回 1 。<p>
 * authenticationManager.generate("bbb", 7); // 时刻 7 时，生成一个 tokenId 为 "bbb" 的新验证码。<p>
 * authenticationManager.renew("aaa", 8); // tokenId 为 "aaa" 的验证码在时刻 7 过期，且 8 >= 7 ，所以时刻 8 的renew 操作被忽略，没有验证码被更新。<p>
 * authenticationManager.renew("bbb", 10); // tokenId 为 "bbb" 的验证码在时刻 10 没有过期，所以 renew 操作会执行，该 token 将在时刻 15 过期。<p>
 * authenticationManager.countUnexpiredTokens(15); // tokenId 为 "bbb" 的验证码在时刻 15 过期，tokenId 为 "aaa" 的验证码在时刻 7 过期，所有验证码均已过期，所以返回 0 。<p>
 * <p>
 * 提示：<p>
 * 1 <= timeToLive <= 10^8<p>
 * 1 <= currentTime <= 10^8<p>
 * 1 <= tokenId.length <= 5<p>
 * tokenId 只包含小写英文字母。<p>
 * 所有 generate 函数的调用都会包含独一无二的 tokenId 值。<p>
 * 所有函数调用中，currentTime 的值 严格递增 。<p>
 * 所有函数的调用次数总共不超过 2000 次。<p>
 * <p>
 * 来源：力扣（LeetCode）<p>
 * 链接：<a href="https://leetcode.cn/problems/design-authentication-manager">1797. 设计一个验证系统</a><p>
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。<p>
 */
public class LeetCode1797 {

    @Test
    public void test01() {
        // 构造 AuthenticationManager ，设置 timeToLive = 5 秒。
        AuthenticationManager authenticationManager = new AuthenticationManager(5);
        // 时刻 1 时，没有验证码的 tokenId 为 "aaa" ，没有验证码被更新。
        authenticationManager.renew("aaa", 1);
        // 时刻 2 时，生成一个 tokenId 为 "aaa" 的新验证码。
        authenticationManager.generate("aaa", 2);
        // 时刻 6 时，只有 tokenId 为 "aaa" 的验证码未过期，所以返回 1 。
        System.out.println(authenticationManager.countUnexpiredTokens(6));
        // 时刻 7 时，生成一个 tokenId 为 "bbb" 的新验证码。
        authenticationManager.generate("bbb", 7);
        // tokenId 为 "aaa" 的验证码在时刻 7 过期，且 8 >= 7 ，所以时刻 8 的renew 操作被忽略，没有验证码被更新。
        authenticationManager.renew("aaa", 8);
        // tokenId 为 "bbb" 的验证码在时刻 10 没有过期，所以 renew 操作会执行，该 token 将在时刻 15 过期。
        authenticationManager.renew("bbb", 10);
        // tokenId 为 "bbb" 的验证码在时刻 15 过期，tokenId 为 "aaa" 的验证码在时刻 7 过期，所有验证码均已过期，所以返回 0 。
        System.out.println(authenticationManager.countUnexpiredTokens(15));
    }
}

class AuthenticationManager {

    int timeToLive;

    Map<String, Integer> map;

    /**
     * 构造 AuthenticationManager 并设置 timeToLive 参数
     *
     * @param timeToLive 过期时间
     */
    public AuthenticationManager(int timeToLive) {
        this.timeToLive = timeToLive;
        map = new HashMap<>(16);
    }

    /**
     * 给定 tokenId ，在当前时间 currentTime 生成一个新的验证码
     *
     * @param tokenId     tokenId
     * @param currentTime currentTime
     */
    public void generate(String tokenId, int currentTime) {
        map.put(tokenId, currentTime);
    }

    /**
     * 将给定 tokenId 且 未过期 的验证码在 currentTime 时刻更新
     * 如果给定 tokenId 对应的验证码不存在或已过期，请你忽略该操作，不会有任何更新操作发生
     *
     * @param tokenId     tokenId
     * @param currentTime currentTime
     */
    public void renew(String tokenId, int currentTime) {
        if (map.containsKey(tokenId) && map.get(tokenId) + timeToLive > currentTime) {
            map.put(tokenId, currentTime);
        }
    }

    /**
     * 请返回在给定 currentTime 时刻，未过期 的验证码数目。
     * 如果一个验证码在时刻 t 过期，且另一个操作恰好在时刻 t 发生（renew 或者 countUnexpiredTokens 操作），过期事件 优先于 其他操作
     *
     * @param currentTime currentTime
     * @return 未过期验证码数目
     */
    public int countUnexpiredTokens(int currentTime) {
        int sum = 0;
        for (Integer value : map.values()) {
            if (value + timeToLive > currentTime) {
                sum++;
            }
        }
        return sum;
    }
}

/**
 * Your AuthenticationManager object will be instantiated and called as such:
 * AuthenticationManager obj = new AuthenticationManager(timeToLive);
 * obj.generate(tokenId,currentTime);
 * obj.renew(tokenId,currentTime);
 * int param_3 = obj.countUnexpiredTokens(currentTime);
 */
